package week_07;

/**
 * 684. 冗余连接
 * 树可以看成是一个连通且 无环 的 无向 图。
 * <p>
 * 给定往一棵 n 个节点 (节点值 1～n) 的树中添加一条边后的图。
 * 添加的边的两个顶点包含在 1 到 n 中间，且这条附加的边不属于树中已存在的边。
 * 图的信息记录于长度为 n 的二维数组 edges ，edges[i] = [ai, bi] 表示图中在 ai 和 bi 之间存在一条边。
 * <p>
 * 请找出一条可以删去的边，删除后可使得剩余部分是一个有着 n 个节点的树。如果有多个答案，则返回数组 edges 中最后出现的边。
 * <p>
 * <p>
 * <p>
 * 示例 1：
 * <p>
 * <p>
 * <p>
 * 输入: edges = [[1,2], [1,3], [2,3]]
 * 输出: [2,3]
 * 示例 2：
 * <p>
 * <p>
 * <p>
 * 输入: edges = [[1,2], [2,3], [3,4], [1,4], [1,5]]
 * 输出: [1,4]
 */
public class FindRedundantConnection_684 {
    int[] fa;

    public int[] findRedundantConnection(int[][] edges) {
        fa = new int[1001];
        for (int i = 0; i <= 1000; i++) fa[i] = i;
        int[] ans = new int[2];
        for (int[] edge : edges) {
            int u = find(edge[0]);
            int v = find(edge[1]);
            if (u == v) {
                ans = edge;
            } else {
                unionSet(u, v);
            }
        }
        return ans;
    }

    public void unionSet(int x, int y) {
        x = find(x);
        y = find(y);
        if (x != y) fa[x] = y;
    }

    public int find(int x) {
        return (x == fa[x]) ? x : (fa[x] = find(fa[x]));
    }
}
